from app import app, db
from models import User

def init_admin():
    with app.app_context():
        # 检查是否已存在管理员账户
        admin = User.query.filter_by(username='admin').first()
        if not admin:
            admin = User(username='admin')
            admin.set_password('admin123')  # 设置初始密码
            db.session.add(admin)
            db.session.commit()
            print("管理员账户创建成功！")
            print("用户名: admin")
            print("密码: admin123")
        else:
            print("管理员账户已存在！如需重置，请先删除数据库重新创建。")

if __name__ == '__main__':
    init_admin() 